#!/bin/bash

# 安装SSH服务
echo "[*] 安装SSH服务..."
apt-get update
apt-get install -y openssh-server

# 备份原始SSH配置文件
echo "[*] 备份原始SSH配置文件..."
cp /etc/ssh/sshd_config /etc/ssh/sshd_config.bak

# 配置SSH允许root登录
echo "[*] 配置SSH允许root登录..."
sed -i 's/#PermitRootLogin prohibit-password/PermitRootLogin yes/g' /etc/ssh/sshd_config
sed -i 's/#PasswordAuthentication yes/PasswordAuthentication yes/g' /etc/ssh/sshd_config

# 重启SSH服务
echo "[*] 重启SSH服务..."
systemctl restart ssh

# 设置SSH开机启动
echo "[*] 设置SSH开机启动..."
systemctl enable ssh

# 获取IP地址和SSH端口
IP_ADDR=$(hostname -I | awk '{print $1}')
SSH_PORT=$(grep -oP '^Port \K\d+' /etc/ssh/sshd_config || echo "22")

# 创建ip.txt文件
echo "[*] 创建连接信息文件..."
cat > ~/ip.txt <<EOF
Kali Linux SSH连接信息:

本地IP地址: $IP_ADDR
SSH端口: $SSH_PORT

SSH连接命令:
ssh root@$IP_ADDR -p $SSH_PORT

连接格式:
ssh [用户名]@[IP地址] -p [端口号]

注意事项:
1. 请确保客户端与服务器在同一网络
2. 首次连接会提示接受密钥指纹
3. 使用完毕后建议修改root密码
EOF

# 显示完成信息和文件内容
echo "[+] SSH配置完成！连接信息已保存到 ~/ip.txt"
echo "----------------------------------------"
cat ~/ip.txt
echo "----------------------------------------"
echo "请确保设置一个强壮的root密码！"